package com.dongnao.network.nio.reactor;

import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author sjy
 * @date 2024年05月11日 1:31
 */
public class CustomThreadPool {

    private static final int NCPUL = Runtime.getRuntime().availableProcessors();

    public static ExecutorService buildPool() {
        ExecutorService pool = new ThreadPoolExecutor(NCPUL, NCPUL * 2, 60L, TimeUnit.SECONDS,
                new LinkedBlockingQueue<>(500), new DefaultThreadFactory());
        return pool;
    }

    private static class DefaultThreadFactory implements ThreadFactory {
        private static final AtomicInteger poolNumber = new AtomicInteger(1);
        private final ThreadGroup group;
        private final AtomicInteger threadNumber = new AtomicInteger(1);
        private final String namePrefix;

        DefaultThreadFactory() {
            SecurityManager s = System.getSecurityManager();
            group = (s != null) ? s.getThreadGroup() : Thread.currentThread().getThreadGroup();
            namePrefix = "pool-" + poolNumber.getAndIncrement() + "-thread-";
        }

        public Thread newThread(Runnable r) {
            Thread t = new Thread(group, r, namePrefix + threadNumber.getAndIncrement(), 0);
            if (t.isDaemon())
                t.setDaemon(false);
            if (t.getPriority() != Thread.NORM_PRIORITY)
                t.setPriority(Thread.NORM_PRIORITY);
            return t;
        }
    }
}
